#include "RetrieveFallenFlag.h"

Vector2 retrieve_grab_flag(Agent* agent, bool* stateChanged);
Vector2 retrieve_come_home(Agent* agent, bool* stateChanged);

RetrieveFallenFlag::~RetrieveFallenFlag(void)
{
}

Vector2 retrieve_grab_flag(Agent* agent, bool* stateChanged) {
	World* world = World::instance();
	string enemy_color = world->other_tanks()[0].color;
	return world->getFlagPos(enemy_color);
}

Vector2 retrieve_come_home(Agent* agent, bool* stateChanged) {
	World* world = World::instance();
	return world->getFlagPos(world->getTeam());
}

void RetrieveFallenFlag::add_waypoint_methods() {
	add_waypoint_method(retrieve_grab_flag);
	add_waypoint_method(retrieve_come_home);
}